import json
import types
# pip install -i https://mirrors.tencent.com/pypi/simple/ --upgrade tencentcloud-sdk-python
from tencentcloud.common import credential
from tencentcloud.common.exception.tencent_cloud_sdk_exception import TencentCloudSDKException
from tencentcloud.common.profile.client_profile import ClientProfile
from tencentcloud.common.profile.http_profile import HttpProfile
from tencentcloud.hunyuan.v20230901 import hunyuan_client, models


def get_answer(secret_id, secret_key, model, address, question):
    try:
        # 实例化一个认证对象，入参需要传入腾讯云账户 SecretId 和 SecretKey，此处还需注意密钥对的保密
        # 代码泄露可能会导致 SecretId 和 SecretKey 泄露，并威胁账号下所有资源的安全性。以下代码示例仅供参考，建议采用更安全的方式来使用密钥，请参见：https://cloud.tencent.com/document/product/1278/85305
        # 密钥可前往官网控制台 https://console.cloud.tencent.com/cam/capi 进行获取
        # cred = credential.Credential("AKID0xolq0BIG6ynT3XsAX1gPvJXaszua5NT", "1oOn0zlQ1aQJJJ9qZ8au2yc8PYZUN9nL")
        cred = credential.Credential(secret_id, secret_key)
        # 实例化一个http选项，可选的，没有特殊需求可以跳过
        httpProfile = HttpProfile()
        httpProfile.endpoint = "hunyuan.tencentcloudapi.com"

        # 实例化一个client选项，可选的，没有特殊需求可以跳过
        clientProfile = ClientProfile()
        clientProfile.httpProfile = httpProfile
        # 实例化要请求产品的client对象,clientProfile是可选的
        client = hunyuan_client.HunyuanClient(cred, address, clientProfile)

        # 实例化一个请求对象,每个接口都会对应一个request对象
        req = models.ChatCompletionsRequest()
        params = {
            "Model": model,
            "MaxToken": 20,
            "Messages": [
                {
                    "Role": "system",
                    "Content": "在选项中找出问题的正确答案，请保证信息的准确，并在回答时保持简洁，不需要任何其他分析反馈。"
                },
                {
                    "Role": "user",
                    "Content": question
                }
            ],
            "Stream": True
        }
        req.from_json_string(json.dumps(params))

        # 返回的resp是一个ChatCompletionsResponse的实例，与请求对象对应
        resp = client.ChatCompletions(req)
        # 输出json格式的字符串回包

        if isinstance(resp, types.GeneratorType):  # 流式响应
            answer = '│ TX：'
            # print('\n│ TX：', end='')
            for event in resp:
                inner_json_str = event['data']
                inner_data_dict = json.loads(inner_json_str)
                choices = inner_data_dict['Choices']
                if choices:
                    # print(choices[0]['Delta']['Content'], end='')
                    answer += choices[0]['Delta']['Content']
            print(answer)
        else:  # 非流式响应
            print(resp)
    except TencentCloudSDKException as err:
        print(err)


# get_answer('AKIDtjbVDpMkxzujT4b5MjNuXIAstmojbXp0', '3AXxT7i8NQpPyU4eEwss2U9mGeCkDlNq', 'hunyuan-pro', 'ap-beijing', '问题：《本草纲目》的作者是谁？选项：[李时珍 张仲景 华佗 孙思邈]')
